6d974a
@@ -274,6 +274,9 @@
public void cancelOperation(OperationHandle opHandle) throws HiveSQLException {
     } else {
       LOG.debug(opHandle + ": Attempting to cancel from state - " + opState);
       operation.cancel(OperationState.CANCELED);
+      if (operation instanceof SQLOperation) {
+        removeSaveSqlOperationDisplay(opHandle);
+      }
     }
   }
 
